from c2rust.roles.baseline import *
from c2rust.roles.rag import *
from c2rust.roles.naive import NaiveRustTranslator, NaiveAddMainFunction

baseline_team = [
    C_Commentator,
    RustTranslator,
    AddMainFunction,
    RustEnhancer,
    RustRunner,
    CompilerMessageAnalyser,
    # TestModifier,
    RustCorrector
]

rag_team = [
    C_Commentator,
    RustTranslator,
    AddMainFunction,
    RustAnalyzer,
    RustTutorialRetriever,
    RustEnhancerRAG,
    RustRunner,
    CompilerMessageAnalyser,
    # TestModifier,
    RustCorrector
]

naive = [
    NaiveRustTranslator,
    NaiveAddMainFunction
]
